package ch02.shili.shili5;
import org.junit.Test;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.transform.OutputKeys;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerConfigurationException;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.io.IOException;

public class BookDom {
    @Test
    public void test() {
        //得到DOM解析器的工厂实例
        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
        //从DOM工厂获得DOM解析器
        try {
            DocumentBuilder db = dbf.newDocumentBuilder();
            //解析XML文档，得到一个Document对象即DOM树
            Document doc = db.parse("src/收藏信息.xml");
            NodeList list = doc.getElementsByTagName("Brand");
            for (int i=0; i<list.getLength(); i++){
                Element brandElement = (Element)list.item(i);
                String brandName = brandElement.getAttribute("name");
                if (brandName.equals("三星")){
                    brandElement.setAttribute("name", "SAMSUNG");//修改属性
                }
            }
            //保存xml文件
            //保存XML文件
            TransformerFactory transformerFactory = TransformerFactory.newInstance();
            try {
                Transformer transformer = transformerFactory.newTransformer();
                DOMSource domSource = new DOMSource(doc);
                //设置编码类型
                transformer.setOutputProperty(OutputKeys.ENCODING, "gb2312");
                StreamResult result = new StreamResult(new FileOutputStream("src/信息收藏.xml"));
                //把DOM树转换为XML文件
            } catch (TransformerConfigurationException e) {
                e.printStackTrace();
            } catch (FileNotFoundException e) {
                e.printStackTrace();
            }
        } catch (ParserConfigurationException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        } catch (SAXException e) {
            e.printStackTrace();
        }
    }
}
